package com.yrot.common.Aspect;

import com.alibaba.fastjson.JSON;
import com.google.gson.Gson;
import com.yrot.admin.entity.SysExpert;
import com.yrot.admin.entity.SysLogEntity;
import com.yrot.admin.service.ISysLogService;
import com.yrot.common.annotation.SysLog;
import com.yrot.common.utils.IPUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.shiro.SecurityUtils;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.LocalVariableTableParameterNameDiscoverer;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;

/**
 * 系统日志，切面处理类
 *
 * @author Mark sunlightcs@gmail.com
 */
@Aspect
@Component
@Slf4j
public class SysLogAspect {
	@Autowired
	private ISysLogService sysLogService;
	
	@Pointcut("@annotation(com.yrot.common.annotation.SysLog)")
	public void logPointCut() { 
		
	}

	@Around("logPointCut()")
	public Object around(ProceedingJoinPoint point) throws Throwable {
		long beginTime = System.currentTimeMillis();
		//执行方法
		Object result = point.proceed();
		//执行时长(毫秒)
		long time = System.currentTimeMillis() - beginTime;

		//保存日志
		saveSysLog(point, time);

		return result;
	}

	private void saveSysLog(ProceedingJoinPoint joinPoint, long time) {
		MethodSignature signature = (MethodSignature) joinPoint.getSignature();
		Method method = signature.getMethod();

		SysLogEntity sysLog = new SysLogEntity();
		SysLog syslog = method.getAnnotation(SysLog.class);
		if(syslog != null){
			//注解上的描述
			sysLog.setOperation(syslog.value());
		}

		//请求的方法名
		String className = joinPoint.getTarget().getClass().getName();
		String methodName = signature.getName();
		sysLog.setMethod(className + "." + methodName + "()");

		//请求的参数
		Object[] args = joinPoint.getArgs();

		try{
			LocalVariableTableParameterNameDiscoverer u = new LocalVariableTableParameterNameDiscoverer();
			String[] paramNames = u.getParameterNames(method);
			if (args != null && paramNames != null) {
				String params = "{";
				for (int i = 0; i < args.length; i++) {
					if(i==(args.length-1)){
						params +=  "\""+paramNames[i] + "\":\"" + args[i]+"\"";
					}else{
						params +=  "\""+paramNames[i] + "\":\"" + args[i]+"\",";
					}
				}
				params +="}";
				sysLog.setParams(params);
			}
			//String params = JSON.toJSONString(args);
			//String params = new Gson().toJson(args[0]);
		}catch (Exception e){
			log.error("参数转换失败");
		}

		//获取request
		HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
		//设置IP地址
		sysLog.setIp(IPUtils.getIpAddr(request));

		//用户名
		SysExpert sysExpert = (SysExpert) SecurityUtils.getSubject().getPrincipal();
		sysLog.setUsername(sysExpert.getExpertUsername());
		sysLog.setName(sysExpert.getExpertName());

		sysLog.setTime(time);
		//保存系统日志
		sysLogService.save(sysLog);
	}
}
